AI unit testing